package com.zut.chat.task;

import com.zut.chat.service.MessageRateLimitService;
import com.zut.chat.service.UserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

/**
 * 定时任务类
 */
@Slf4j
@Component
public class ScheduledTasks {

    @Autowired
    private UserService userService;

    @Autowired
    private MessageRateLimitService messageRateLimitService;

    /**
     * 每5分钟检查并自动解除过期的禁言
     */
    @Scheduled(fixedRate = 300000) // 5分钟 = 300000毫秒
    public void checkAndUnbanExpiredUsers() {
        try {
            userService.checkAndUnbanExpiredUsers();
            log.debug("执行定时任务：检查并解除过期禁言");
        } catch (Exception e) {
            log.error("执行定时任务失败：检查并解除过期禁言", e);
        }
    }

    /**
     * 每小时清理过期的用户消息频率记录
     */
    @Scheduled(fixedRate = 3600000) // 1小时 = 3600000毫秒
    public void cleanupExpiredRateRecords() {
        try {
            messageRateLimitService.cleanupExpiredRecords();
            log.debug("执行定时任务：清理过期的用户消息频率记录");
        } catch (Exception e) {
            log.error("执行定时任务失败：清理过期的用户消息频率记录", e);
        }
    }

    /**
     * 每天凌晨2点清理过期的用户消息频率记录（更彻底的清理）
     */
    @Scheduled(cron = "0 0 2 * * ?")
    public void dailyCleanupExpiredRecords() {
        try {
            messageRateLimitService.cleanupExpiredRecords();
            log.info("执行每日定时任务：清理过期的用户消息频率记录");
        } catch (Exception e) {
            log.error("执行每日定时任务失败：清理过期的用户消息频率记录", e);
        }
    }
} 